# -*- coding: utf-8 -*-
"""
@Time    : 2025/2/12 18:40 
@Author  : ZhangShenao 
@File    : 2.embedding.py 
@Desc    : Embedding API
"""

import os

import dotenv
from openai import OpenAI

# 创建OpenAI客户端
dotenv.load_dotenv()
client = OpenAI(base_url=os.getenv("OPENAI_API_BASE"))

# 调用Embedding API
resp = client.embeddings.create(
    model="text-embedding-3-small",  # 指定Embedding模型
    input="hello world",  # 指定文本内容
    dimensions=512,  # 指定向量维度
    encoding_format="float"  # 指定格式为浮点数
)

# 生成向量
embeddings = resp.data[0].embedding

# 打印结果
print(len(embeddings))  # 512
print(embeddings[:10])
